import cv2
import json

# 读取图像文件
image_path = '1.jpg'
image = cv2.imread(image_path)

# 在图像上绘制矩形框（示例：目标检测中的一个对象）
box_coordinates = [(100, 100), (200, 200)]  # 两个点分别表示左上角和右下角坐标
color = (255, 0, 0)  # 蓝色
thickness = 2
annotated_image = cv2.rectangle(image.copy(), box_coordinates[0], box_coordinates[1], color, thickness)

# 保存标注后的图像
cv2.imwrite('annotated_image.jpg', annotated_image)

# 保存标注数据（示例：保存矩形框的坐标）
annotation_data = {
    'image_path': image_path,
    'annotations': [
        {
            'class': 'object',
            'box_coordinates': box_coordinates
        }
    ]
}

# 将标注数据保存为JSON文件
with open('annotation_data.json', 'w') as f:
    json.dump(annotation_data, f)

print("图像标注和标注数据保存完成。")
